package auth

import (
	"beegoProject10/models/auth"
	"beegoProject10/utils"
	"fmt"
	"github.com/astaxie/beego"
	"github.com/astaxie/beego/orm"
	"math"
	"strings"
	"time"
)

type RoleController struct {
	beego.Controller
}

func (c *RoleController) List() {

	// 数据库查询并传入前端
	o := orm.NewOrm()
	qs := o.QueryTable("sys_role")

	roles := []auth.Role{}
	qs.Filter("is_delete", 0).All(&roles)

	/*
		分页逻辑 ⬇️⬇️⬇️⬇️⬇️⬇️⬇️⬇️⬇️⬇️⬇️⬇️⬇️⬇️⬇️⬇️⬇️⬇️⬇️⬇️⬇️⬇️⬇️
	*/

	// 每页显示数据-4条 (pagePerNum)
	pagePerNum := 5

	// 当前页 (currentPage)
	currentPage, err := c.GetInt("page")
	// 说明没有获取到当前页
	if err != nil {
		currentPage = 1 // 所以默认显示第一页
	}

	// 增加查询：
	offsetNum := pagePerNum * (currentPage - 1)

	// 总数据量
	count, _ := qs.Filter("is_delete", 0).Count()
	// 用户详细信息查询
	qs.Filter("is_delete", 0).Limit(pagePerNum).Offset(offsetNum).All(&roles)

	// 总页数 (count) =  总数据量 / 每页数据量   注：需要转换成int类型
	countPage := int(math.Ceil(float64(count) / float64(pagePerNum)))

	// 判断上一页
	prePage := 1
	if currentPage == 1 {
		prePage = 1
	} else if currentPage > 1 {
		// 上一页
		prePage = currentPage - 1
	}

	// 下一页
	nextPage := 1
	// 如果当前页小于总页数
	if currentPage < countPage {
		// 下一页 就等于 当前页+1
		nextPage = currentPage + 1
		// 如果当前页面大于等于总页数
	} else if currentPage >= countPage {
		// 下一页 就等于 当前页
		nextPage = currentPage
	}

	// 引用倒入到函数page.go
	page_map := utils.Paginator(currentPage, pagePerNum, count)

	c.Data["prePage"] = prePage         /* 上一页*/
	c.Data["nextPage"] = nextPage       /* 下一页 */
	c.Data["currentPage"] = currentPage /* 当前页 */
	c.Data["countPage"] = countPage
	c.Data["count"] = count
	c.Data["page_map"] = page_map
	/*
		分页逻辑 ⬆️⬆️⬆️⬆️⬆️⬆️⬆️⬆️⬆️⬆️⬆️⬆️⬆️⬆️⬆️⬆️⬆️⬆️⬆️⬆️⬆️⬆️⬆️⬆️⬆️
	*/

	c.Data["roles"] = roles
	c.TplName = "auth/role-list.html"
}

func (c *RoleController) ToAdd() {

	c.TplName = "auth/role-add.html"

}

func (c *RoleController) DoAdd() {

	// 1
	role_name := c.GetString("role_name")
	desc := c.GetString("desc")
	is_active, _ := c.GetInt("is_active")

	// 3
	role := auth.Role{
		RoleName:   role_name,
		Desc:       desc,
		IsActive:   is_active,
		CreateTime: time.Now(),
	}

	// 2
	o := orm.NewOrm()
	_, err := o.Insert(&role)

	// 4
	message_map := map[string]interface{}{}

	// 5
	if err != nil {
		message_map["code"] = 10001
		message_map["msg"] = "添加数据出错，请重新添加"
	} else {
		message_map["code"] = 200
		message_map["msg"] = "添加成功"
	}

	// 6
	c.Data["json"] = message_map
	c.ServeJSON()
}

func (c *RoleController) ToRoleUser() {

	id, _ := c.GetInt("role_id")
	fmt.Println("获取当前角色ID： ======> ", id)

	// 数据库查询初id对应的权限
	role := auth.Role{} // 单个ID
	o := orm.NewOrm()
	o.QueryTable("sys_role").Filter("id", id).One(&role)
	fmt.Println("权限信息: ----------> ", role)

	// 用户信息查出
	users := []auth.User{} // 多个属于切片
	o.QueryTable("sys_user").Filter("is_delete", 0).Filter("is_active", 1).All(&users)
	fmt.Println("用户列表: ----------> ", users)

	c.Data["role"] = role
	c.Data["users"] = users
	c.TplName = "auth/role-user-add.html"

}

func (c *RoleController) DoRoleUser() {

	// 1
	roleId, _ := c.GetInt("role_id")
	userIds := c.GetString("user_ids")
	fmt.Println("role_id: ", roleId)
	fmt.Println("userIds: ", userIds)

	/*         9.59【权限管理模块】角色上配置用户二         */
	// ++++++++++  如果有双引号则需要左右切割  ++++++++++
	// 2 切割userIds到双引号，从第二个到长度最后
	// newUserIds := userIds[1 : len(userIds)-1]
	// 切割前 "10,12,13,16,20"
	// 切割完  10,12,13,16,20

	// 3 按照逗号切割成数组
	userIdArr := strings.Split(userIds, ",")
	fmt.Println("userIdArr: ", userIdArr)

	// 6
	o := orm.NewOrm()

	// 4 角色Id对象
	role := auth.Role{
		Id: roleId,
	}

	// 5 用户数据
	for _, userId := range userIdArr {

		// 用户Id对象
		user := auth.User{
			Id: utils.StrToInt(userId), // string类型转换，编写int转换 utils
		}

		// 7
		m2m := o.QueryM2M(&role, "User") // 通过role去查询 User(必须和模型名一直)
		m2m.Add(user)                    // 多对多循环插入，插入需要和模型名字一致user
	}

	c.Data["json"] = map[string]interface{}{
		"code": 200,
		"msg":  "添加成功",
	}
	c.ServeJSON()

}
